from bert_model import BertConfig
import mindspore.common.dtype as mstype



def bert_net_cfg(batch_size):
    bert_config=BertConfig(
    batch_size=batch_size,
    seq_length=128,
    vocab_size=3191,
    hidden_size=768,
    num_hidden_layers=12,
    num_attention_heads=12,
    intermediate_size=3072,
    hidden_act="gelu",
    hidden_dropout_prob=0.1,
    attention_probs_dropout_prob=0.1,
    max_position_embeddings=512,
    type_vocab_size=2,
    initializer_range=0.02,
    use_relative_positions=False,
    input_mask_from_dataset=True,
    token_type_ids_from_dataset=True,
    dtype=mstype.float32,
    compute_type=mstype.float16,)
    return bert_config


